class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        unordered_map<int, int> hash;   // 统计前缀和为 0 和它的下标
        hash[0] = -1;   // 默认有一个前缀和为 0 的情况

        int sum = 0, ret = 0;
        for(int i = 0; i < nums.size(); i++)
        {
            sum += nums[i] == 0 ? -1 : 1;   // 计算当前位置的前缀和
            if(hash.count(sum))
                ret = max(ret, i - hash[sum]);
            else
                hash[sum] = i;
        }
        return ret;
    }
};